package Intermediate_algorithm.Math;


import org.junit.Test;

/*
x 的平方根
给你一个非负整数 x ，计算并返回x的 算术平方根 。
由于返回类型是整数，结果只保留 整数部分 ，小数部分将被 舍去 。
注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5) 或者 x ** 0.5 。

示例 1：
输入：x = 4
输出：2
示例 2：
输入：x = 8
输出：2
解释：8 的算术平方根是 2.82842..., 由于返回类型是整数，小数部分将被舍去。

作者：LeetCode
链接：https://leetcode.cn/leetbook/read/binary-search/xe9cog/
 */
public class _05x的平方根 {

    @Test
    public void test() {
        System.out.println(mySqrt(2147395599));
    }

    //WA
    //超时
    public int mySqrt(int x) {
        if (x == 0 || x == 1)
            return x;
        for (int i = x/2; i > 0; i--) {
            if (i * i <= x)
                return i;
        }
        return -1;
    }

    //二分查找
    public static int mySqrt2(int x) {
        if (x == 0 || x == 1) {
            return x;
        }
        int left = 1,right = x / 2;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            long midVal = (long)mid * mid;
            if (midVal == x ) {
                return mid;
            }else if (midVal < x) {
                left = mid + 1;
            }else {
                right = mid - 1;
            }
        }
        return right;
    }

    //官解：二分查找
    public int mySqrt3(int x) {
        int l = 0, r = x, ans = -1;
        while (l <= r) {
            int mid = l + (r - l) / 2;
            if ((long) mid * mid <= x) {
                ans = mid;
                l = mid + 1;
            } else {
                r = mid - 1;
            }
        }
        return ans;
    }

}
